Home
last modified time | relevance | path

Searched defs:in_shape (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbcast_ops.cc44 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
93 const TensorShape in_shape = ctx->InputShape(i); in Compile() local
Dconcat_op.cc76 const TensorShape& in_shape = shapes[i]; in Compile() local
Dlrn_ops.cc38 const TensorShape in_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dsplit_op_test.cc32 TensorShape in_shape(chunk_size); in MakeGraph() local
Dsplit_v_op_test.cc60 TensorShape in_shape(total_size); in MakeGraph() local
Dscoped_allocator_ops_test.cc261 void BuildNodeDef(const TensorShape& in_shape, DataType dtype, in BuildNodeDef()
277 void MakeOp(const TensorShape& in_shape, DataType dtype, const string& name, in MakeOp()
Dcudnn_pooling_gpu.cc41 const auto in_shape = tensor_in.shape(); in Compute() local
Dfractional_avg_pool_op.cc260 TensorShape in_shape; in Compute() local
Dbias_op.cc426 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcudnn_determinism_test.py55 def _random_out_op(self, in_shape, filter_shape): argument
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_transpose_to_reshape.cc27 std::vector<int> in_shape) { in TransposeAffectsMemoryOrder()
Dmerge_reshape_into_preceding_transpose.cc66 std::vector<int> in_shape = input_array.shape().dims(); in ReshapeToTranspose() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcopy_test.cc245 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc453 TensorShapeProto in_shape; in CreateTRTNode() local
Dconvert_nodes.cc252 auto in_shape = input_params.at(in_port); in GetInputProperties() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc138 TensorShape in_shape, TensorShape out_shape) { in MakeGatherTestCase()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis_test.cc2121 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2264 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
Dtuple_points_to_analysis_test.cc931 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
Dalgebraic_simplifier_test.cc3383 auto in_shape = make_shape(in_dims, options.input_minor_to_major_layout); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1653 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
1727 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
1833 Tensor in_shape = test::AsTensor<int32>( in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc2216 std::vector<int> in_shape = input_array.shape().dims(); in ReshapeIsEquivalentToTranspose() local