Home
last modified time | relevance | path

Searched refs:base_shape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt65 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.float32
66 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.uint8
67 DISABLED_concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.int32
68 DISABLED_concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=1,type=tf.int64
69 concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.float32
70 concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.uint8
71 DISABLED_concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.int32
72 DISABLED_concat/concat_axis=0,base_shape=[3,4],num_tensors=1,type=tf.int64
73 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=2,type=tf.float32
74 concat/concat_axis=0,base_shape=[1,3,4,3],num_tensors=2,type=tf.uint8
[all …]
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Dsampler.py362 base_shape = array_ops.shape(base_next_inputs)
364 indices=where_sampling, updates=sampled_next_inputs, shape=base_shape)
368 shape=base_shape))
477 base_shape = array_ops.shape(base_next_inputs)
479 indices=where_sampling, updates=sampled_next_inputs, shape=base_shape)
483 shape=base_shape))
Dhelper.py402 base_shape = array_ops.shape(base_next_inputs)
405 shape=base_shape)
408 shape=base_shape))
538 base_shape = array_ops.shape(base_next_inputs)
541 shape=base_shape)
544 shape=base_shape))
/external/tensorflow/tensorflow/lite/kernels/
Dadd_test.cc361 const std::vector<int> base_shape = {2, 3, 1, 2}; in QuantizedWithMixedBroadcast() local
378 QuantizedAddOpModel model_fixture({tensor_type, base_shape, -3.f, 3.f}, in QuantizedWithMixedBroadcast()
396 {tensor_type, base_shape, -3.f, 3.f}, in QuantizedWithMixedBroadcast()
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Dvirtual_batchnorm_impl.py104 base_shape = tensor_shape.TensorShape(_get_shape(tensor_list[0]))
106 base_shape.assert_is_compatible_with(_get_shape(tensor))
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc159 StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, in InferWindowOutputShape() argument
163 if (window.dimensions_size() != base_shape.rank()) { in InferWindowOutputShape()
166 window.dimensions_size(), base_shape.rank()); in InferWindowOutputShape()
200 if (base_shape.is_dynamic_dimension(i) && in InferWindowOutputShape()
208 ShapeUtil::GetDimension(base_shape, i), dim.base_dilation()); in InferWindowOutputShape()
216 output_is_dynamic[i] = base_shape.is_dynamic_dimension(i); in InferWindowOutputShape()
1775 Shape base_shape = in InferConvolveShape() local
1779 InferWindowOutputShape(base_shape, window, lhs.element_type(), in InferConvolveShape()
Dhlo_evaluator_typed_visitor.h2824 const Shape& window_shape, const Window& window, const Shape& base_shape, in IterateThroughWindow() argument
2827 const int64 rank = base_shape.rank(); in IterateThroughWindow()
2845 if (base_index[i] < 0 || base_index[i] >= base_shape.dimensions(i)) { in IterateThroughWindow()