HomeSort by relevance Sort by last modified time
    Searched refs:GetIntAttr (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.i 129 bool GetIntAttr(PyObject* o, const char* field, int64* result) {
675 if (!GetIntAttr(item, "edge_padding_low", &edge_padding_low)
676 || !GetIntAttr(item, "edge_padding_high", &edge_padding_high)
677 || !GetIntAttr(item, "interior_padding", &interior_padding)) {
701 if (!GetIntAttr($input, "input_batch_dimension", &value)) {
706 if (!GetIntAttr($input, "input_feature_dimension", &value)) {
711 if (!GetIntAttr($input, "output_batch_dimension", &value)) {
716 if (!GetIntAttr($input, "output_feature_dimension", &value)) {
721 if (!GetIntAttr($input, "kernel_output_feature_dimension", &value)) {
726 if (!GetIntAttr($input, "kernel_input_feature_dimension", &value))
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
lrn_op_test.cc 40 int GetIntAttr(const string& name) {
60 const int64 depth_radius = GetIntAttr("depth_radius");
  /external/tensorflow/tensorflow/contrib/lite/toco/
import_tensorflow.cc 77 int GetIntAttr(const NodeDef& node, const string& attr_name) {
476 op->block_size = GetIntAttr(node, "block_size");
491 op->block_size = GetIntAttr(node, "block_size");
809 const int num_split = GetIntAttr(node, "num_split");
    [all...]

Completed in 104 milliseconds