| /external/tensorflow/tensorflow/lite/kernels/ |
| pad.cc | 41 constant_values = GetOptionalInputTensor(context, node, 2);
43 constant_values = nullptr;
59 const TfLiteTensor* constant_values; member in struct:tflite::ops::builtin::pad::PadContext
102 if (op_context.constant_values != nullptr) {
104 op_context.constant_values->type);
123 if (op_context.constant_values != nullptr) {
124 // Ensure that constant_values is a scalar.
125 TF_LITE_ENSURE_EQ(context, NumElements(op_context.constant_values), 1);
167 float pad_value = op_context.constant_values == nullptr
169 : *GetTensorData<float>(op_context.constant_values);
[all...] |
| pad_test.cc | 74 std::initializer_list<int> paddings, T1 constant_values,
80 this->AddConstInput(GetTensorType<T1>(), {constant_values}, {1});
92 const TensorData& constant_values,
97 this->constant_values_ = this->AddInput(constant_values);
136 RegularInputOuput constant_values,
141 GetTensorType<RegularInputOuput>(), {constant_values}, {1});
150 const TensorData& constant_values,
154 this->constant_values_ = this->AddInput(constant_values);
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| pad_op_test.py | 34 def _npPad(self, inp, paddings, mode, constant_values=0):
37 return np.pad(inp, paddings, mode=mode, constant_values=constant_values)
64 mode="constant", constant_values=1))
86 def _testPad(self, np_inputs, paddings, mode, constant_values):
88 constant_values=constant_values)
91 constant_values=constant_values)
96 def _testGradient(self, x, a, mode, constant_values)
[all...] |
| /external/deqp/external/openglcts/modules/glesext/geometry_shader/ |
| esextcGeometryShaderConstantVariables.cpp | 218 glw::GLint constant_values[n_varyings] = { 0 }; local
221 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_INPUT_COMPONENTS, &constant_values[index]);
225 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_OUTPUT_COMPONENTS, &constant_values[index]);
229 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_TEXTURE_IMAGE_UNITS, &constant_values[index]);
233 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_OUTPUT_VERTICES, &constant_values[index]);
237 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_TOTAL_OUTPUT_COMPONENTS, &constant_values[index]);
241 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_UNIFORM_COMPONENTS, &constant_values[index]);
245 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_ATOMIC_COUNTERS, &constant_values[index]);
249 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_ATOMIC_COUNTER_BUFFERS, &constant_values[index]);
253 gl.getIntegerv(m_glExtTokens.MAX_GEOMETRY_IMAGE_UNIFORMS, &constant_values[index])
[all...] |
| /external/tensorflow/tensorflow/python/ops/signal/ |
| dct_ops.py | 122 constant_values=n2)
134 constant_values=n2)
|
| shape_ops.py | 142 signal = array_ops.pad(signal, paddings, constant_values=pad_value)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pad_op.cc | 70 // PadV2 added a "constant_values" input that indicates the pad value.
71 xla::XlaOp constant_values; variable
74 ctx, TensorShapeUtils::IsScalar(ctx->InputShape("constant_values")),
75 errors::InvalidArgument("constant_values must be a scalar."));
76 ctx->SetOutput(0, xla::Pad(input, ctx->Input("constant_values"), config));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| pad_op.cc | 75 const Tensor& constant_values = context->input(2); variable
77 context, TensorShapeUtils::IsScalar(constant_values.shape()),
78 errors::InvalidArgument("constant_values must be a scalar. Found: ",
79 constant_values.shape().DebugString()));
283 .HostMemory("constant_values"), \
290 .HostMemory("constant_values"), \
347 .HostMemory("constant_values"), \
354 .HostMemory("constant_values"), \
386 .HostMemory("constant_values")
395 .HostMemory("constant_values")
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| stateful_random_ops.py | 118 constant_values=0)
|
| array_ops.py | [all...] |
| ctc_ops.py | 468 constant_values=math_ops.log(0.0))
[all...] |
| /external/tensorflow/tensorflow/contrib/resampler/python/ops/ |
| resampler_ops_test.py | 58 data = np.lib.pad(data, 1, "constant", constant_values=0)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| helper.py | 69 constant_values=1))
|
| sampler.py | 710 constant_values=1))
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| batching.py | 298 value, array_ops.stack([left, right], 1), constant_values=padding_value)
|
| /external/deqp/external/openglcts/modules/gl/ |
| gl4cShaderSubroutineTests.cpp | 4636 const Utils::vec4<glw::GLfloat> constant_values[] = { Utils::vec4<glw::GLfloat>(1, 2, 3, 4), local
4718 const Utils::vec4<glw::GLfloat> constant_values[] = { Utils::vec4<glw::GLfloat>(1, 2, 3, 4), local
[all...] |
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| util.py | [all...] |
| /external/v8/src/heap/ |
| factory.h | 203 ElementsKind elements_kind, Handle<FixedArrayBase> constant_values);
[all...] |
| factory.cc | 243 ElementsKind elements_kind, Handle<FixedArrayBase> constant_values) {
248 result->set_constant_elements(*constant_values);
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| binary_ops_test.py | [all...] |
| /external/tensorflow/tensorflow/lite/testing/ |
| generate_examples.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
